/**
 * This file is part of aion-emu <aion-emu.com>.
 *
 *  aion-emu is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  aion-emu is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with aion-emu.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.aionengine.gameserver.dataholders;

import com.aionengine.gameserver.model.DialogAction;
import com.aionengine.gameserver.model.TribeClass;
import com.aionengine.gameserver.model.gameobjects.Npc;
import com.aionengine.gameserver.model.gameobjects.player.Player;
import com.aionengine.gameserver.model.templates.npc.NpcTemplate;
import com.aionengine.gameserver.utils.PacketSendUtility;
import gnu.trove.map.hash.TIntObjectHashMap;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.Unmarshaller;
import javax.xml.bind.annotation.*;
import java.io.File;
import java.util.*;

/**
 * This is a container holding and serving all {@link NpcTemplate} instances.<br>
 * Briefly: Every {@link Npc} instance represents some class of NPCs among which each have the same id, name, items,
 * statistics. Data for such NPC class is defined in {@link NpcTemplate} and is uniquely identified by npc id.
 *
 * @author Luno
 */
@XmlRootElement(name = "npc_templates")
@XmlAccessorType(XmlAccessType.FIELD)
public class NpcData extends ReloadableData {

    @XmlElement(name = "npc_template")
    private List<NpcTemplate> npcs;

    /**
     * A map containing all npc templates
     */
    @XmlTransient
    private TIntObjectHashMap<NpcTemplate> npcData = new TIntObjectHashMap<NpcTemplate>();

    @XmlTransient
    private HashSet<TribeClass> unusedTribes = new HashSet<TribeClass>();

    void afterUnmarshal(Unmarshaller u, Object parent) {
        unusedTribes.addAll(Arrays.asList(TribeClass.values()));
        for (NpcTemplate npc : npcs) {
            npcData.put(npc.getTemplateId(), npc);
            if (npc.getTribe() != null)
                unusedTribes.remove(npc.getTribe());
            if (npc.getFuncDialogIds() != null) {
                for (Integer dialogId : npc.getFuncDialogIds()) {
                    DialogAction dialogAction = DialogAction.getActionByDialogId(dialogId);
                    if (dialogAction == null) {
                        log.warn("Missing dialog action " + dialogId + " for Npc " + npc.getTemplateId());
                    }
                }
            }
        }
        npcs.clear();
        npcs = null;

        Iterator<TribeClass> iter = unusedTribes.iterator();
        if (unusedTribes.size() > 0) {
            while (iter.hasNext())
                iter.next().setUsed(false);
            unusedTribes.clear();
        }
        unusedTribes = null;
    }

    public int size() {
        return npcData.size();
    }

    /**
     * /** Returns an {@link NpcTemplate} object with given id.
     *
     * @param id id of NPC
     * @return NpcTemplate object containing data about NPC with that id.
     */
    public NpcTemplate getNpcTemplate(int id) {
        return npcData.get(id);
    }

    /**
     * @return the npcData
     */
    public TIntObjectHashMap<NpcTemplate> getNpcData() {
        return npcData;
    }

    @Override
    public void reload(Player admin) {
        File dir = new File("./data/static_data/npcs");
        try {
            JAXBContext jc = JAXBContext.newInstance(StaticData.class);
            Unmarshaller un = jc.createUnmarshaller();
            un.setSchema(getSchema("./data/static_data/static_data.xsd"));
            List<NpcTemplate> newTemplates = new ArrayList<NpcTemplate>();
            for (File file : listFiles(dir, true)) {
                NpcData data = (NpcData) un.unmarshal(file);
                if (data != null && data.getData() != null)
                    newTemplates.addAll(data.getData());
            }
            DataManager.NPC_DATA.setData(newTemplates);
        } catch (Exception e) {
            PacketSendUtility.sendMessage(admin, "Npc reload failed!");
            log.error("Npc reload failed!", e);
        } finally {
            PacketSendUtility.sendMessage(admin, "Npc reload Success! Total loaded: " + DataManager.NPC_DATA.size());
        }
    }

    protected List<NpcTemplate> getData() {
        return npcs;
    }

    @SuppressWarnings("unchecked")
    protected void setData(List<?> templates) {
        this.npcs = (List<NpcTemplate>) templates;
        afterUnmarshal(null, null);
    }
}
